Cassava Leaf Disease Classification
In this competition, we are trying to identify common diseases of cassava crops using data science and machine learning. Previous methods of disease detection require farmers to solicit the help of government-funded agricultural experts to visually inspect and diagnose the plants. This suffers from being labor-intensive, low-supply and costly. Instead, it would be preferred if an automated pipeline based on mobile-quality photos of the cassava leafs could be developed.
This competition provides a farmer-crowdsourced dataset, labeled by experts at the National Crops Resources Research Institute (NaCRRI).
In this kernel, I will present a quick EDA.
import numpy as np
import pandas as pd
import seaborn as sns
import albumentations as A
import matplotlib.pyplot as plt
import os, gc, cv2, random, warnings, math, sys, json, pprint, pdb
import tensorflow as tf
from tensorflow.keras import backend as K
import tensorflow_hub as hub
from sklearn.model_selection import train_test_split
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
warnings.simplefilter('ignore')
Tip: Adding seed helps reproduce results. Setting debug parameter wil run the model on smaller number of epochs to validate the architecture.
SEED = 16
DEBUG = True #@param {type:"boolean"}
os.environ['PYTHONHASHSEED'] = str(SEED)
np.random.seed(SEED)
tf.random.set_seed(SEED)
from google.colab import drive
drive.mount('/content/gdrive', force_remount=True)
dataset_path = '/content/gdrive/MyDrive/1_AUSTIN CHEN/Data Scientist/Datasets/cassava-leaf-disease-classification'
os.chdir(dataset_path)
os.listdir(dataset_path)
df = pd.read_csv(dataset_path + '/train.csv')
df.head()
Check how many images are available in the training dataset and also check if each item in the training set are unique
print(f"There are {len(df)} train images")
len(df.image_id) == len(df.image_id.unique())
(df.label.value_counts(normalize=True) * 100).plot.barh(figsize = (8, 5))
df['filename'] = df['image_id'].map(lambda x : dataset_path + '/train_images/' + x)
df = df.drop(columns = ['image_id'])
df = df.sample(frac=1).reset_index(drop=True)
df.head()
if DEBUG:
_, df = train_test_split(
df,
test_size = 0.1,
random_state=SEED,
shuffle=True,
stratify=df['label'])
with open(dataset_path + '/label_num_to_disease_map.json') as file:
id2label = json.loads(file.read())
id2label
In this case, we have 5 labels (4 diseases and healthy):
- Cassava Bacterial Blight (CBB)
- Cassava Brown Streak Disease (CBSD)
- Cassava Green Mottle (CGM)
- Cassava Mosaic Disease (CMD)
- Healthy
In this case label 3, Cassava Mosaic Disease (CMD) is the most common label. This imbalance may have to be addressed with a weighted loss function or oversampling. I might try this in a future iteration of this kernel or in a new kernel.
Let's check an example image to see what it looks like
from PIL import Image
img = Image.open(df[df.label==3]['filename'].iloc[0])
width, height = img.size
print(f"Width: {width}, Height: {height}")
img
SELECT = ("efficientnet_b3", 300) #param ["(\"efficientnet_b4\", 380)", "(\"efficientnet_b2\", 260)"] {type:"raw", allow-input: true}
BASE_MODEL, SIZE = SELECT
BASE_MODEL_URL = "https://tfhub.dev/google/imagenet/{}/feature_vector/4".format(BASE_MODEL)
BATCH_SIZE = 32 #param {type:"integer"}
IMG_SIZE = (SIZE, SIZE)
print("Using {} with input size {}".format(BASE_MODEL, SIZE))
Load data
After my quick and rough EDA, let's load the PIL Image to a Numpy array, so we can move on to data augmentation.
In fastai, they have item_tfms and batch_tfms defined for their data loader API. The item transforms performs a fairly large crop to 224 and also apply other standard augmentations (in aug_tranforms) at the batch level on the GPU. The batch size is set to 32 here.
Important: Since we are using built-in generator, it takes label as string
df['label'] = df['label'].astype('str')
train_df, valid_df = train_test_split(
df
,test_size = 0.2
,random_state=SEED
,shuffle = True
,stratify = df['label']
)
train_ds = tf.data.Dataset.from_tensor_slices(
(train_df.filename.values,train_df.label.values))
adapt_ds = tf.data.Dataset.from_tensor_slices(
(train_df.filename.values,train_df.label.values))
valid_ds = tf.data.Dataset.from_tensor_slices(
(valid_df.filename.values, valid_df.label.values))
for x,y in valid_ds.take(3):
print(x, y)
AUTOTUNE = tf.data.experimental.AUTOTUNE
Important: At this point, you may have noticed that I have not used any kind of normalization or rescaling. I recently discovered that there is Normalization layer included in Keras’ pretrained EfficientNet, as mentioned here.
def process_train(filename, label):
img = tf.io.read_file(filename)
img = tf.image.decode_jpeg(img, channels=3)
img = tf.image.random_brightness(img, 0.3)
img = tf.image.random_flip_left_right(img, seed=None)
img = tf.image.random_crop(img, size=[*IMG_SIZE, 3])
return img, label
def process_adapt(filename, label):
img = tf.io.read_file(filename)
img = tf.image.decode_jpeg(img, channels=3)
img = tf.keras.layers.experimental.preprocessing.Rescaling(1.0 / 255)(img)
return img, label
def process_valid(filename, label):
# load the raw data from the file as a string
img = tf.io.read_file(filename)
img = tf.image.decode_jpeg(img, channels=3)
img = tf.image.resize(img, [*IMG_SIZE])
return img, label
train_ds = train_ds.map(process_train, num_parallel_calls=AUTOTUNE)
adapt_ds = adapt_ds.map(process_adapt, num_parallel_calls=AUTOTUNE)
valid_ds = valid_ds.map(process_valid, num_parallel_calls=AUTOTUNE)
def show_images(ds):
_,axs = plt.subplots(4,6,figsize=(24,16))
for ((x, y), ax) in zip(ds.take(24), axs.flatten()):
ax.imshow(x.numpy().astype(np.uint8))
ax.set_title(np.argmax(y))
ax.axis('off')
show_images(train_ds)
show_images(valid_ds)
Note: I was shuffing the validation set which is a bug
train_ds_batch = (train_ds
.shuffle(buffer_size=1000)
.batch(BATCH_SIZE)
.prefetch(buffer_size=AUTOTUNE))
adapt_ds_batch = (adapt_ds
.shuffle(buffer_size=1000)
.batch(BATCH_SIZE)
.prefetch(buffer_size=AUTOTUNE))
valid_ds_batch = (valid_ds
#.shuffle(buffer_size=1000)
.batch(BATCH_SIZE*2)
.prefetch(buffer_size=AUTOTUNE))
image_batch, label_batch = next(iter(train_ds_batch))
plt.figure(figsize=(10, 10))
for i in range(16):
ax = plt.subplot(4, 4, i + 1)
plt.imshow(image_batch[i].numpy().astype("uint8"))
label = label_batch[i].numpy()
plt.title(label)
plt.axis("off")
data_augmentation = tf.keras.Sequential(
[
tf.keras.layers.experimental.preprocessing.RandomCrop(*IMG_SIZE),
tf.keras.layers.experimental.preprocessing.RandomFlip("horizontal_and_vertical"),
tf.keras.layers.experimental.preprocessing.RandomRotation(0.25),
tf.keras.layers.experimental.preprocessing.RandomZoom((-0.2, 0)),
tf.keras.layers.experimental.preprocessing.RandomContrast((0.2,0.2))
]
)
tf.keras.layers.experimental.preprocessing.RandomCrop()
plt.figure(figsize=(10, 10))
for i in range(16):
augmented_images = data_augmentation(image_batch)
ax = plt.subplot(4, 4, i + 1)
plt.imshow(augmented_images[i].numpy().astype("uint8"))
label = label_batch[i].numpy()
plt.title(label)
plt.axis("off")
from tensorflow.keras.applications import EfficientNetB3
!wget https://storage.googleapis.com/keras-applications/efficientnetb3_notop.h5
do_fine_tuning = False #@param {type:"boolean"}
efficientnet = EfficientNetB3(
weights=dataset_path + "/efficientnetb3_notop.h5",
include_top=False,
input_shape=(*IMG_SIZE, 3),
drop_connect_rate=0.4)
efficientnet.summary()
model = tf.keras.Sequential([
tf.keras.layers.InputLayer(input_shape=(*IMG_SIZE, 3)),
data_augmentation,
efficientnet,
tf.keras.layers.GlobalAveragePooling2D(),
tf.keras.layers.Dense(256),
tf.keras.layers.BatchNormalization(),
tf.keras.layers.Activation('relu'),
tf.keras.layers.Dropout(0.3),
tf.keras.layers.Dense(5, activation='softmax')
])
model.summary()
The 3rd layer of the Efficient is the Normalization layer, which can be tuned to our new dataset instead of imagenet. Be patient on this one, it does take a bit of time as we're going through the entire training set.
%%time
model.get_layer('efficientnetb3').get_layer('normalization_2').adapt(adapt_ds_batch)
adapt_ds_batch.element_spec
def build_my_model(base_model, optimizer, loss='categorical_crossentropy', metrics = ['categorical_accuracy']):
my_model = Sequential()
my_model.add(base_model)
my_model.add(GlobalAveragePooling2D())
my_model.add(Dense(256))
my_model.add(BatchNormalization())
my_model.add(Activation('relu'))
my_model.add(Dropout(0.3))
my_model.add(Dense(5, activation='softmax'))
my_model.compile(
optimizer=optimizer,
loss=CategoricalCrossentropy(label_smoothing=0.05),
metrics=metrics
)
return my_model
def load_pretrained_model(weights_path, drop_connect,
target_size_dim, layers_to_unfreeze=5):
model = EfficientNetB3(
weights=None,
include_top=False,
input_shape=(target_size_dim, target_size_dim, 3),
drop_connect_rate=0.4
)
model.load_weights(weights_path)
model.trainable = True
# for layer in model.layers[-layers_to_unfreeze:]:
# if not isinstance(layer, tf.keras.layers.BatchNormalization):
# layer.trainable = True
if DEBUG:
for layer in model.layers:
print(layer.name, layer.trainable)
return model
1% Better Everyday
https://www.kaggle.com/frlemarchand/efficientnet-aug-tf-keras-for-cassava-diseases https://www.kaggle.com/harveenchadha/efficientnetb3-keras-tf2-baseline-training
todos
- Find out the intuition and the difference between
item_tfmandbatch_tfm - Customize my own data generator as fastai creates their
Dataloader - Prepare a special dataset that will be fed to the Normalization layer. The
EfficientnetB3provided bytf.kerasincludes an out-of-the-box Normalization layer fit onto theimagenetdataset. Therefore, we can pull that layer and use theadaptfunction to retrain it to the Cassava Disease dataset. - The 3rd layer of the Efficientnet is the Normalization layer, which can be tuned to our new dataset instead of
imagenet. Be patient on this one, it does take a bit of time we're going through the entire training set.
done
- Try out the
data_generatorand thedata_frame_iterator - Removing normalizaiton step in generator since in EfficientNet, normalization is done within the model itself and the model expects input in the range of [0,255]
def albu_transforms_train(data_resize):
return A.Compose([
A.ToFloat(),
A.Resize(data_resize, data_resize),
], p=1.)
# For Validation
def albu_transforms_valid(data_resize):
return A.Compose([
A.ToFloat(),
A.Resize(data_resize, data_resize),
], p=1.)
def CutMix(image, label, DIM, PROBABILITY = 1.0):
# input image - is a batch of images of size [n,dim,dim,3] not a single image of [dim,dim,3]
# output - a batch of images with cutmix applied
CLASSES = 5
imgs = []; labs = []
for j in range(len(image)):
# DO CUTMIX WITH PROBABILITY DEFINED ABOVE
P = tf.cast( tf.random.uniform([],0,1)<=PROBABILITY, tf.int32)
# CHOOSE RANDOM IMAGE TO CUTMIX WITH
k = tf.cast( tf.random.uniform([],0,len(image)),tf.int32)
# CHOOSE RANDOM LOCATION
x = tf.cast( tf.random.uniform([],0,DIM),tf.int32)
y = tf.cast( tf.random.uniform([],0,DIM),tf.int32)
b = tf.random.uniform([],0,1) # this is beta dist with alpha=1.0
WIDTH = tf.cast( DIM * tf.math.sqrt(1-b),tf.int32) * P
ya = tf.math.maximum(0,y-WIDTH//2)
yb = tf.math.minimum(DIM,y+WIDTH//2)
xa = tf.math.maximum(0,x-WIDTH//2)
xb = tf.math.minimum(DIM,x+WIDTH//2)
# MAKE CUTMIX IMAGE
one = image[j,ya:yb,0:xa,:]
two = image[k,ya:yb,xa:xb,:]
three = image[j,ya:yb,xb:DIM,:]
middle = tf.concat([one,two,three],axis=1)
img = tf.concat([image[j,0:ya,:,:],middle,image[j,yb:DIM,:,:]],axis=0)
imgs.append(img)
# MAKE CUTMIX LABEL
a = tf.cast(WIDTH*WIDTH/DIM/DIM,tf.float32)
labs.append((1-a)*label[j] + a*label[k])
# RESHAPE HACK SO TPU COMPILER KNOWS SHAPE OF OUTPUT TENSOR (maybe use Python typing instead?)
image2 = tf.reshape(tf.stack(imgs),(len(image),DIM,DIM,3))
label2 = tf.reshape(tf.stack(labs),(len(image),CLASSES))
return image2,label2
def MixUp(image, label, DIM, PROBABILITY = 1.0):
# input image - is a batch of images of size [n,dim,dim,3] not a single image of [dim,dim,3]
# output - a batch of images with mixup applied
CLASSES = 5
imgs = []; labs = []
for j in range(len(image)):
# DO MIXUP WITH PROBABILITY DEFINED ABOVE
P = tf.cast( tf.random.uniform([],0,1)<=PROBABILITY, tf.float32)
# CHOOSE RANDOM
k = tf.cast( tf.random.uniform([],0,len(image)),tf.int32)
a = tf.random.uniform([],0,1)*P # this is beta dist with alpha=1.0
# MAKE MIXUP IMAGE
img1 = image[j,]
img2 = image[k,]
imgs.append((1-a)*img1 + a*img2)
# MAKE CUTMIX LABEL
labs.append((1-a)*label[j] + a*label[k])
# RESHAPE HACK SO TPU COMPILER KNOWS SHAPE OF OUTPUT TENSOR (maybe use Python typing instead?)
image2 = tf.reshape(tf.stack(imgs),(len(image),DIM,DIM,3))
label2 = tf.reshape(tf.stack(labs),(len(image),CLASSES))
return image2,label2